#!/usr/bin/python3
# -*- coding:utf-8 -*-
# __author__ == taoyulong2018@gmail.com
# __time__ == 2023/1/10 17:05
# ===========================================
#       题目名称： 4. 寻找两个正序数组的中位数
#       题目地址： https://leetcode.cn/problems/median-of-two-sorted-arrays/
#       题目描述： https://note.youdao.com/s/LWYgPSzT
# ===========================================


class Solution:
    def findMedianSortedArrays(self, nums1, nums2):
        """
            寻找两个正序数组的中位数
            @param nums1: 列表1
            @param nums2: 列表2
        """
        nums1.extend(nums2)
        nums1.sort()
        if len(nums1) % 2 == 0:     # 偶数
            return (nums1[int(len(nums1)/2)] + nums1[int(len(nums1)/2 - 1)]) / 2.0
        else:                       # 奇数
            return nums1[int(len(nums1)/2)]


if __name__ == '__main__':
    s = Solution()
    print(s.findMedianSortedArrays([1, 3], [2, ]))      # 2
    print(s.findMedianSortedArrays([1, 2], [3, 4]))     # 2.5








